Ternary operator in python